home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / test / test_dumbdbm.pyo (.txt) < prev    next >
Python Compiled Bytecode  |  2005-10-18  |  5KB  |  179 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyo (Python 2.4)
  3.  
  4. '''Test script for the dumbdbm module
  5.    Original by Roger E. Masse
  6. '''
  7. import os
  8. import unittest
  9. import dumbdbm
  10. from test import test_support
  11. _fname = test_support.TESTFN
  12.  
  13. def _delete_files():
  14.     for ext in [
  15.         '.dir',
  16.         '.dat',
  17.         '.bak']:
  18.         
  19.         try:
  20.             os.unlink(_fname + ext)
  21.         continue
  22.         except OSError:
  23.             continue
  24.         
  25.  
  26.     
  27.  
  28.  
  29. class DumbDBMTestCase(unittest.TestCase):
  30.     _dict = {
  31.         '0': '',
  32.         'a': 'Python:',
  33.         'b': 'Programming',
  34.         'c': 'the',
  35.         'd': 'way',
  36.         'f': 'Guido',
  37.         'g': 'intended' }
  38.     
  39.     def __init__(self, *args):
  40.         unittest.TestCase.__init__(self, *args)
  41.  
  42.     
  43.     def test_dumbdbm_creation(self):
  44.         f = dumbdbm.open(_fname, 'c')
  45.         self.assertEqual(f.keys(), [])
  46.         for key in self._dict:
  47.             f[key] = self._dict[key]
  48.         
  49.         self.read_helper(f)
  50.         f.close()
  51.  
  52.     
  53.     def test_close_twice(self):
  54.         f = dumbdbm.open(_fname)
  55.         f['a'] = 'b'
  56.         self.assertEqual(f['a'], 'b')
  57.         f.close()
  58.         f.close()
  59.  
  60.     
  61.     def test_dumbdbm_modification(self):
  62.         self.init_db()
  63.         f = dumbdbm.open(_fname, 'w')
  64.         self._dict['g'] = f['g'] = 'indented'
  65.         self.read_helper(f)
  66.         f.close()
  67.  
  68.     
  69.     def test_dumbdbm_read(self):
  70.         self.init_db()
  71.         f = dumbdbm.open(_fname, 'r')
  72.         self.read_helper(f)
  73.         f.close()
  74.  
  75.     
  76.     def test_dumbdbm_keys(self):
  77.         self.init_db()
  78.         f = dumbdbm.open(_fname)
  79.         keys = self.keys_helper(f)
  80.         f.close()
  81.  
  82.     
  83.     def test_write_write_read(self):
  84.         f = dumbdbm.open(_fname)
  85.         f['1'] = 'hello'
  86.         f['1'] = 'hello2'
  87.         f.close()
  88.         f = dumbdbm.open(_fname)
  89.         self.assertEqual(f['1'], 'hello2')
  90.         f.close()
  91.  
  92.     
  93.     def test_line_endings(self):
  94.         f = dumbdbm.open(_fname)
  95.         f['1'] = 'hello'
  96.         f['2'] = 'hello2'
  97.         f.close()
  98.         data = open(_fname + '.dir').read()
  99.         data = data.replace('\n', '\r\n')
  100.         open(_fname + '.dir', 'wb').write(data)
  101.         f = dumbdbm.open(_fname)
  102.         self.assertEqual(f['1'], 'hello')
  103.         self.assertEqual(f['2'], 'hello2')
  104.  
  105.     
  106.     def read_helper(self, f):
  107.         keys = self.keys_helper(f)
  108.         for key in self._dict:
  109.             self.assertEqual(self._dict[key], f[key])
  110.         
  111.  
  112.     
  113.     def init_db(self):
  114.         f = dumbdbm.open(_fname, 'w')
  115.         for k in self._dict:
  116.             f[k] = self._dict[k]
  117.         
  118.         f.close()
  119.  
  120.     
  121.     def keys_helper(self, f):
  122.         keys = f.keys()
  123.         keys.sort()
  124.         dkeys = self._dict.keys()
  125.         dkeys.sort()
  126.         self.assertEqual(keys, dkeys)
  127.         return keys
  128.  
  129.     
  130.     def test_random(self):
  131.         import random as random
  132.         d = { }
  133.         for dummy in range(5):
  134.             f = dumbdbm.open(_fname)
  135.             for dummy in range(100):
  136.                 k = random.choice('abcdefghijklm')
  137.                 if random.random() < 0.20000000000000001:
  138.                     if k in d:
  139.                         del d[k]
  140.                         del f[k]
  141.                     
  142.                 k in d
  143.                 v = random.choice('abc') * random.randrange(10000)
  144.                 d[k] = v
  145.                 f[k] = v
  146.                 self.assertEqual(f[k], v)
  147.             
  148.             f.close()
  149.             f = dumbdbm.open(_fname)
  150.             expected = d.items()
  151.             expected.sort()
  152.             got = f.items()
  153.             got.sort()
  154.             self.assertEqual(expected, got)
  155.             f.close()
  156.         
  157.  
  158.     
  159.     def tearDown(self):
  160.         _delete_files()
  161.  
  162.     
  163.     def setUp(self):
  164.         _delete_files()
  165.  
  166.  
  167.  
  168. def test_main():
  169.     
  170.     try:
  171.         test_support.run_unittest(DumbDBMTestCase)
  172.     finally:
  173.         _delete_files()
  174.  
  175.  
  176. if __name__ == '__main__':
  177.     test_main()
  178.  
  179.